package com.example.bishe.demos.web;

import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.tencentcloudapi.common.AbstractModel;

import com.tencentcloudapi.common.Credential;
import com.tencentcloudapi.common.profile.ClientProfile;
import com.tencentcloudapi.common.profile.HttpProfile;
import com.tencentcloudapi.common.exception.TencentCloudSDKException;
import com.tencentcloudapi.iai.v20200303.IaiClient;
import com.tencentcloudapi.iai.v20200303.models.*;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.*;

import java.util.HashMap;
import java.util.Map;

@CrossOrigin(origins = "http://localhost:8080")
@RestController
public class TXAI_Face_recognition {
    @PostMapping("/face-recognition")
    public ResponseEntity<Map<String, Object>> faceRecognition(@RequestBody Map<String, String> request) {
        Map<String, Object> response = new HashMap<>();
        ObjectMapper mapper = new ObjectMapper();
        try {
            Credential cred = new Credential("AKIDrZAZmlCalmdJn6PaYKvKdkdQ2Y3vfDnF", "ITSEQVNG7pUzQrOybIuKYZMVUM1ThgTB");
            IaiClient client = new IaiClient(cred, "ap-beijing", new ClientProfile());

            SearchFacesRequest req = new SearchFacesRequest();
            req.setGroupIds(new String[]{"renjie"});
            req.setImage(request.get("image"));  // 直接传入base64
            req.setMaxFaceNum(1L);
            req.setMinFaceSize(80L);
            req.setMaxPersonNum(5L);
            req.setNeedPersonInfo(0L);
            req.setQualityControl(0L);
            req.setNeedRotateDetection(0L);
            SearchFacesResponse resp = client.SearchFaces(req);
      String jsonStr = SearchFacesRequest.toJsonString(resp);
            try {
                JsonNode rootNode = mapper.readTree(jsonStr);
                JsonNode scoreNode = rootNode.path("Results").get(0).path("Candidates").get(0).path("Score");
                response.put("data", scoreNode.asText());
            } catch (JsonProcessingException e) {
                throw new RuntimeException("jsone: " + e);
            }



        } catch (TencentCloudSDKException e) {
            response.put("code", 500);
            response.put("error", e.toString());
        }
        return ResponseEntity.ok(response);
    }
//    public static void main(String [] args) {
//        try{
//            // 实例化一个认证对象，入参需要传入腾讯云账户 SecretId 和 SecretKey，此处还需注意密钥对的保密
//            // 代码泄露可能会导致 SecretId 和 SecretKey 泄露，并威胁账号下所有资源的安全性
//            // 以下代码示例仅供参考，建议采用更安全的方式来使用密钥
//            // 请参见：https://cloud.tencent.com/document/product/1278/85305
//            // 密钥可前往官网控制台 https://console.cloud.tencent.com/cam/capi 进行获取
//            Credential cred = new Credential("AKIDrZAZmlCalmdJn6PaYKvKdkdQ2Y3vfDnF", "ITSEQVNG7pUzQrOybIuKYZMVUM1ThgTB");
//            // 实例化一个http选项，可选的，没有特殊需求可以跳过
//            HttpProfile httpProfile = new HttpProfile();
//            httpProfile.setEndpoint("iai.tencentcloudapi.com");
//            // 实例化一个client选项，可选的，没有特殊需求可以跳过
//            ClientProfile clientProfile = new ClientProfile();
//            clientProfile.setHttpProfile(httpProfile);
//            // 实例化要请求产品的client对象,clientProfile是可选的
//            IaiClient client = new IaiClient(cred, "ap-beijing", clientProfile);
//            // 实例化一个请求对象,每个接口都会对应一个request对象
//            SearchFacesRequest req = new SearchFacesRequest();
//            String[] groupIds1 = {"renjie"};
//            req.setGroupIds(groupIds1);
//
//            req.setImage("");
//            req.setMaxFaceNum(1L);
//            req.setMinFaceSize(80L);
//            req.setMaxPersonNum(5L);
//            req.setNeedPersonInfo(0L);
//            req.setQualityControl(0L);
//            req.setNeedRotateDetection(0L);
//            // 返回的resp是一个SearchFacesResponse的实例，与请求对象对应
//            SearchFacesResponse resp = client.SearchFaces(req);
//            // 输出json格式的字符串回包
//            System.out.println(AbstractModel.toJsonString(resp));
//        } catch (TencentCloudSDKException e) {
//            System.out.println(e.toString());
//        }
//    }
}